package dp.最优解;

/**
 * todo
 *
 * @author songZiHao
 * @version 1.0.0
 * @ClassName 买卖股票的最佳时机含冷冻期309.java
 * @createTime 2024年03月19日 17:49:00
 */
public class 买卖股票的最佳时机含冷冻期309 {
	/*
    定义dp[i][k][0/1]表示第i天的上限交易次数k的时候，未持有股票和持有股票的最大金额
    因为有冷冻期
    dp[i][k][0]=Math.max(dp[i-1][k-1][1]+num[i],dp[i-1][k][0]);
    dp[i][k][1]=Math.max(dp[i-2][k-1][0]-num[i],dp[i-1][k-1][1]);
    k无限次可以省略
    baseCase：
     dp[0][0] = 0;
     dp[0][1] = -num[0];
      dp[i][0] = Math.max(dp[i - 1][1] + prices[i], dp[i - 1][0]);
     dp[i][1] = Math.max(-prices[i], dp[i - 1][1]);
     */
	public int maxProfit(int[] prices) {
		int length = prices.length;
		int[][] dp = new int[length][2];
		dp[0][0] = 0;
		dp[0][1] = 0;

		for (int i = 0; i < length; i++) {
			//第一条
			if (i == 0) {
				dp[0][0] = 0;
				dp[0][1] = -prices[0];
				continue;
			}
			if (i == 1) {
				dp[i][0] = Math.max(dp[i - 1][1] + prices[i], dp[i - 1][0]);
				dp[i][1] = Math.max(-prices[i], dp[i - 1][1]);
				continue;
			}
			dp[i][0] = Math.max(dp[i - 1][1] + prices[i], dp[i - 1][0]);
			dp[i][1] = Math.max(dp[i - 2][0] - prices[i], dp[i - 1][1]);
		}
		return dp[length - 1][0];
	}
}
